package com.liangtian.mdm.dm.common.util;

import java.util.*;

/**
 * Description: 分组集合时使用
 */
public class GroupUtil {


    /**
     * 分组依据接口，用于集合分组时,获取分组依据
     *
     * @param <T>
     */
    public interface GroupBy<T> {

        T groupby(Object obj);
    }

    /**
     * 将给定的集合按条件进行分组
     *
     * @param colls 要分组的集合
     * @param gb    分组依据
     * @return
     */
    public static final <T extends Comparable<T>, D> Map<T, List<D>> group(Collection<D> colls, GroupBy<T> gb) {
        //分组集合不能为空
        if (colls == null || colls.isEmpty()) {
            System.out.println("分组集合不能为空!");
            return null;
        }
        //分组依据不能为null
        if (gb == null) {
            System.out.println("分组依据不能为空!");
            return null;
        }
        Iterator<D> iter = colls.iterator();
//        Map<T, List<D >> map = new HashMap<>();
        LinkedHashMap<T, List<D>> map = new LinkedHashMap<>();
        while (iter.hasNext()) {
            D d = iter.next();
            T t = gb.groupby(d);
            if (map.containsKey(t)) {
                map.get(t).add(d);
            } else {
                List<D> list = new ArrayList<>();
                list.add(d);
                map.put(t, list);
            }
        }
        return map;
    }

    /**
     * 将给定的集合按条件进行分组；
     * 并对组的key、value排序
     *
     * @param colls 要分组的集合
     * @param gb    分组依据
     * @return
     */
    public static final <T extends Comparable<T>, D extends Comparable<D>> TreeMap<T, List<D>> groupOrder(Collection<D> colls, GroupBy<T> gb) {

        Map<T, List<D>> group = group(colls, gb);
        TreeMap<T, List<D>> map = new TreeMap<>();
        for (T t : group.keySet()) {
            List<D> v = group.get(t);
            Collections.sort(v);
            map.put(t, v);
        }
        return map;
    }

    public static final <T extends Comparable<T>, D> List<Map<T, D>> groupTree(Collection<D> colls, GroupBy<T> gb, String childKey) {

        List<Map<T, D>> list = new ArrayList<>();

        return list;
    }
}
